Deep Fusion Memory: A Chunk-Level Parametric Memory for Large Language Models
深度融合记忆:大语言模型的块级参数化记忆
🧭 参数化记忆 · 隐藏层深度融合
LUMIA Lab, Shanghai Jiao Tong University · Shanghai Artificial Intelligence Laboratory
提出块级参数化记忆DFM,训练数据构造检索量降低32倍,并深度融合进隐藏状态,无需在线检索
📌 工作参考价值(5 点)
- ### 对电商 / 广告 / Agent 的可借鉴点
- 在电商导购、广告文案或客服 Agent 中,需要给 LLM 注入领域知识(商品库、活动政策、广告规范)。DFM 提供一种冻结主干的参数化记忆方案:推理时无显式检索步骤,延迟低,比 RAG 更适合高 QPS 线上场景;可训练轻量 memory predictor + fusion 模块,把领域知识压缩进隐藏状态。
- 块级记忆监督是关键技巧:用 chunk embedding 作为预测目标,而非 token 级目标,训练数据构造的检索 query 可减少 32×、检索条目减少 2048×。业务数据量大、离线检索成本敏感时,优先考虑把知识切块并做块级记忆,能大幅降低知识注入成本。
- 门控 cross-attention 融合比输出层插值更深,能让知识影响中间表征;工程上可只插入少数层,冻结原模型,避免灾难性遗忘。适合给已有的推荐/Agent 底座模型做领域适配而不破坏通用能力。
- 可迁移到实时交互场景:让 memory predictor 根据对话或行为前文预测 chunk 级记忆,动态融合到解码过程,省去显式检索模块,简化架构并降低运维复杂度。